package com.huike.clues.service.impl;

import java.io.IOException;
import java.io.InputStream;
import java.net.URLEncoder;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.UUID;

import javax.servlet.http.HttpServletResponse;

import com.alibaba.excel.EasyExcel;
import com.huike.clues.domain.Clue;
import io.minio.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import com.huike.clues.service.ISysFileService;
import com.huike.common.config.MinioConfig;
import com.huike.common.core.domain.AjaxResult;

import lombok.extern.slf4j.Slf4j;

@Service
@Slf4j
public class SysFileServiceImpl implements ISysFileService {

    @Autowired
    MinioConfig minioConfig;

    /**
     * 文件上传至Minio
     * 使用try catch finally进行上传
     * finally里进行资源的回收
     */
    @Override
    public AjaxResult upload(MultipartFile file) {
        InputStream inputStream = null;
        //创建Minio的连接对象
        MinioClient minioClient = getClient();
        String bucketName = minioConfig.getBucketName();
        try {
            inputStream = file.getInputStream();
            //基于官网的内容，判断文件存储的桶是否存在 如果桶不存在就创建桶
            boolean exists = minioClient.bucketExists(BucketExistsArgs.builder().bucket("huike-crm").build());
            if (!exists){  //不存在 则创建一个新的
                minioClient.makeBucket(MakeBucketArgs.builder().bucket("huike-crm").build());
            }else {
                log.info("bucketName '桶名' already Exists ");
            }
            //TODO 补全这部分代码
            /**
             * ================================操作文件================================
             * 思路：我们上传的文件是:合同.pdf
             * 那么我们应该上传到配置的bucket内 我们配置的bucketName是huike-crm
             * 那么我们存在桶里的文件应该是什么样的 也叫“合同.pdf”吗?
             * 应该按照上传的年月日进行区分
             * 举例：2021-05-05日进行上传的
             * 那么存在桶里的路径应该是
             * huike-crm/2021/05/05/这个目录下
             * 而对于同一个文件，存在重名问题，所以我们应该利用UUID生成一个新的文件名，并拼接上 .pdf 作为文件后缀
             * 那么完整的路径就是 huike-crm/2021/05/05/uuid.pdf
             *
             * 如果上述思路你无法理解，那么就直接存放在桶内生成uuid+.pdf即可
             * 即：huike-crm/uuid.pdf
             */
            //TODO 基于上述逻辑补全代码
            /**
             * 构建返回结果集
             */
                AjaxResult ajax = AjaxResult.success();
            /**
             * 封装需要的数据进行返回
             */
            String fileName = file.getOriginalFilename();
            String objectName = new SimpleDateFormat("yyyy/MM/dd/").format(new Date())
                    + UUID.randomUUID().toString().replaceAll("-", "")
                    + fileName.substring(fileName.lastIndexOf("."));
            //2. 文件上传  参数校验 和分片上传
            PutObjectArgs objectArgs = PutObjectArgs.builder().object(objectName)
                    .bucket(bucketName)
                    .contentType(file.getContentType())
                    .stream(file.getInputStream(), file.getSize(), -1).build();
            minioClient.putObject(objectArgs);
            //3. 构建返回结果集,将文件名(fileName)和访问路径(url)返回
            ajax.put("fileName", "/" + bucketName + "/" + objectName);
            ajax.put("url", "http://" + minioConfig.getEndpoint() + ":" + minioConfig.getPort()
                    + "/" + minioConfig.getBucketName() + "/" + objectName);
            //4. 将结果返回
            return ajax;
        } catch (Exception e) {
            e.printStackTrace();
            return AjaxResult.error("上传失败");
        } finally {
            //防止内存泄漏
            if (inputStream != null) {
                try {
                    inputStream.close(); // 关闭流
                } catch (IOException e) {
                    log.debug("inputStream close IOException:" + e.getMessage());
                }
            }
        }
    }

    /**
     * 文件下载
     * @param
     * @return
     */
    @Override
    public void download(HttpServletResponse response) {
        ArrayList<Clue> Clues = new ArrayList<>();
        response.setCharacterEncoding("utf-8");
        // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
        try {
            //String fileName = URLEncoder.encode("订单数据", "UTF-8").replaceAll("\\+", "%20");
            response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + "clues" + ".xlsx");
            EasyExcel.write(response.getOutputStream(), Clue.class).sheet("订单信息").doWrite(Clues);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }


    /**
     * 免费提供一个获取Minio连接的方法
     * 获取Minio连接
     *
     * @return
     */
    private MinioClient getClient() {
        MinioClient minioClient =
                MinioClient.builder()
                        .endpoint("http://" + minioConfig.getEndpoint() + ":" + minioConfig.getPort())
                        .credentials(minioConfig.getAccessKey(), minioConfig.getSecretKey())
                        .build();
        return minioClient;
    }
}
